#!/usr/bin/env python
# -*- coding: utf-8 -*-
"""
taichi_dither 性能测试脚本
该脚本用于测试不同抖动算法和taichi的性能表现
"""

import argparse
from taichi_dither import DitherTester

def main():
    # 解析命令行参数
    parser = argparse.ArgumentParser(description='测试抖动算法的性能')
    parser.add_argument('--image', type=str, default='test.jpg', help='测试图像路径')
    parser.add_argument('--output', type=str, default='test_results', help='输出结果目录')
    parser.add_argument('--fast', action='store_true', help='快速测试模式，减少测试用例')
    parser.add_argument('--no-cache', action='store_true', help='禁用Taichi编译缓存')
    args = parser.parse_args()
    
    # 创建测试器
    tester = DitherTester(
        args.image, 
        args.output, 
        fast_mode=args.fast,
        use_cache=not args.no_cache
    )
    
    # 运行测试
    print("开始性能测试...")
    tester.run_all_tests()
    
    # 生成报告
    tester.generate_report()
    
    # 生成图表
    tester.generate_charts()
    
    print("\n测试完成！")

if __name__ == "__main__":
    main() 